from ai.data_processing.feature_extractor import FeatureExtractor

def test_extract_features():
    extractor = FeatureExtractor()
    features = extractor.extract_features(y=some_audio_data, sr=44100)
    assert 'chroma' in features
    assert 'mfcc' in features